#pragma once

#include <unistd.h>
#include <fcntl.h>
#include <cstdlib>

#define NON_BLOCK_ERR 1
void SetNonBlockOrDie(int sockfd)
{
    int f1 = fcntl(sockfd, F_GETFL);
    if (f1 < 0)
        exit(NON_BLOCK_ERR);
    fcntl(sockfd, F_SETFL, f1 | O_NONBLOCK);
}